// 异步日志缓冲区

#ifndef __M_BUF_H__
#define __M_BUF_H__
#include "util.hpp"
#include <vector>
#include <cassert>

namespace gjclog
{
    #define DEFAULT_BUFFER_SIZE (1*1024*1024)      // 缓冲区默认大小
    #define THRESHOLD_BUFFER_SIZE (8*1024*1024)    // 缓冲区大小阈值
    #define INCREMENT_BUFFER_SIZE (1*1024*1024)    // 缓冲区线性增长大小
    class Buffer
    {
        public:
            Buffer()
            : _buffer(DEFAULT_BUFFER_SIZE), _writer_idx(0), _reader_idx(0)
            {}
            
            // 向缓冲区写入数据
            void push(const char* data, size_t len)
            {
                // 1. 判断缓冲区剩余空间是否充足，不够则扩容
                ensureEnoughSize(len);
                // 2. 将数据拷贝进缓冲区
                std::copy(data, data+len, &_buffer[_writer_idx]);
                // 3. 将写入位置向后偏移len
                moveWriter(len);
            }
            // 获取可写数据长度
            size_t writeAbleSize()
            {
                // 扩容：总是可写
                // 固定大小缓冲区：提供此功能
                return (_buffer.size() - _writer_idx);
            }
            // 获取可读数据的起始地址
            const char* begin()
            {
                return (&_buffer[_reader_idx]);
            }
            // 获取可读数据的长度
            size_t readAbleSzie()
            {
                // 不是环形队列，双缓冲区，写满直接交换
                return (_writer_idx - _reader_idx);
            }
            // 对读指针进行向后偏移
            void moveReader(size_t len)
            {
                assert(len <= readAbleSzie());
                _reader_idx += len; 
            }
            // 重置读写位置指针，初始化缓冲区
            void reset()
            {
                _writer_idx = 0;
                _reader_idx = 0;
            }
            // 对缓冲区Buffer进行交换
            void swap(Buffer &buffer)
            {
                _buffer.swap(buffer._buffer);
                std::swap(_writer_idx, buffer._writer_idx);
                std::swap(_reader_idx, buffer._reader_idx);
            }
            // 判断缓冲区是否为空
            bool empty()
            {
                return (_reader_idx == _writer_idx);
            }
        private:
            // 缓冲区扩容
            void ensureEnoughSize(size_t len)
            {
                if(len <= writeAbleSize())   // 剩余空间充足，不用扩容
                {
                    return;
                }
                size_t new_size = 0;
                if(_buffer.size() < THRESHOLD_BUFFER_SIZE)  // 缓冲区空间小于阈值，翻倍扩容
                {
                    new_size = _buffer.size() * 2 + len;
                }
                else    // 缓冲区去大小大于等于阈值，线性扩容
                {
                    new_size = _buffer.size() + INCREMENT_BUFFER_SIZE + len;
                }
                // 扩容时+len是为了保证扩容后大小大于len
                _buffer.resize(new_size);
            }
            // 对写指针进行向后偏移
            void moveWriter(size_t len)
            {
                assert((len + _writer_idx) <= _buffer.size());
                _writer_idx += len;
            }
        private:
            std::vector<char> _buffer;  // 缓冲区
            size_t _reader_idx;   // 当前可读数据指针（下标）
            size_t _writer_idx; // 当前可写数据指针（下标）
    };
}

#endif